package com.example.demo.utils;

import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.io.IOException;
import java.util.UUID;

public class FileUploadUtil {

    // 定义指定的文件夹名
    private static final String UPLOAD_FOLDER = "src/main/resources/uploads";

    /**
     * 上传文件到指定文件夹，并返回文件的存储路径
     * @param file 上传的文件
     * @return 文件的存储路径，如果上传失败则返回 null
     */
    public static String uploadFile(MultipartFile file) {
        if (file.isEmpty()) {
            return null;
        }
        try {
            // 获取当前项目的根目录
            String projectPath = System.getProperty("user.dir");
            // 拼接指定的文件夹路径
            String uploadPath = projectPath + File.separator + UPLOAD_FOLDER;
            // 创建指定的文件夹
            File uploadDir = new File(uploadPath);
            if (!uploadDir.exists()) {
                uploadDir.mkdirs();
            }
            // 生成唯一的文件名
            String originalFilename = file.getOriginalFilename();
            String fileExtension = originalFilename.substring(originalFilename.lastIndexOf("."));
            String uniqueFileName = UUID.randomUUID().toString() + fileExtension;
            // 拼接文件的存储路径
            String filePath = uploadPath + File.separator + uniqueFileName;
            // 保存文件到指定的文件夹中
            File dest = new File(filePath);
            file.transferTo(dest);
            return filePath;
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }
}